@gfabbris This is ready for review.

I don't see how it is possible to get values from baseline with any of the three functions, when only a key fragment is given. Can key_fragment and strict also be implemented for getRunData and getRunDataValue?

There is a stream keyword that is used to pick the "baseline" stream. Default is "primary". Example is in the unit tests: https://github.com/BCDA-APS/apstools/blob/502-find_devices/tests/test_utils.py#L348-L376

The real situation is that the databroker v2 search is slow compared with the v1 interface. For now, rely on the v1 interface.

Time some test functions to compare:

import databroker
catalog_name = "training"  # on my test workstation

%%timeit
cat = databroker.catalog[catalog_name]
for idx in range(-20, 0):
    run = cat.v1[idx]
    uid = run.start["uid"]
    # if  "baseline" in run.stream_names:
    #     bl = run.table(stream_name="baseline")
    #     print(idx, uid)


%%timeit
cat = databroker.catalog[catalog_name]
for idx in range(-20, 0):
    run = cat.v2[idx]
    uid = run.metadata["start"]["uid"]
    # if hasattr(run, "baseline"):
    #    baseline = run.baseline.read()
    #    print(idx, uid)

mean ± std. dev. of 7 runs, 1 loop each

In commented, the baseline data is not actually read from databroker. Roughly twice as long to use the v2 interface instead of the v1 interface.

In [9]: databroker.__version__
Out[9]: '1.2.2'

The database is on a LAN MongoDB server.

getStreamValues() is hoisted from 4ID Polar's lookup_position(). It returns a pandas DataFrame (instead of printing to console). If the object is not assigned to a variable, it prints with nice format.

Needs unit testing and could benefit from a lower-case match instead of explicit case match.

@gfabbris , @strempfer , @CrI3 ; This is ready for review.

@strempfer Great catch. Turns out there are no unit tests yet for getStreamValues(). That's an oversight I will need to fix. As a result, you should get a proper error if db is not defined and no default database can be found. I suspect you actually have a database. It should be found by default.

@strempfer : Added unit tests. If you do not specify a db argument and you have multiple catalog configurations available, then a ValueError will be raised since apstools.utils.getDefaultCatalog() cannot decide which catalog configuration to select. There is now a unit test for this.